package com.mlh.binarytree.BFS;

import com.mlh.binarytree.TreeNode;

import java.util.ArrayList;
import java.util.Deque;
import java.util.LinkedList;
import java.util.List;

/**
 * @author 缪林辉
 * @date 2024/3/26 10:23
 * @DESCRIPTION
 */
//给定一个非空二叉树的根节点 root , 以数组的形式返回每一层节点的平均值。与实际答案相差 10-5 以内的答案可以被接受。
public class AverageOfLevels {
    public List<Double> method1(TreeNode root) {
        if(root==null){return new ArrayList<>();}
        List<Double>res=new ArrayList<>();
        Deque<TreeNode> treeNodes = new LinkedList<>();
        treeNodes.add(root);
        while(!treeNodes.isEmpty()){
            int len=treeNodes.size();
            int lenCopy=len;
            double temp=0;
            while (len>0){
                TreeNode node=treeNodes.poll();
                len--;
                temp+=node.val;
                if(node.left!=null){
                    treeNodes.add(node.left);
                }
                if(node.right!=null){
                    treeNodes.add(node.right);
                }
            }
            res.add(temp/lenCopy);
        }
        return res;
    }
}
